"""
直接使用模型检查预测概率分布
"""
import sys
sys.path.append('/Volumes/d/MyUnderGraduate/ModelRecordOuputDetectionSystem/safety_api')

from validators.ml_content_safety import MLContentSafety

# 初始化验证器
validator = MLContentSafety(
    model_path="../models/safety_classifier_v2",
    threshold=0.5
)

# 测试样本
samples = [
    "抓虫大战，人人有责",
    "好久都没有见过那么干净的白人班级了，一个黑乐色都没有",
    "估计已经成了非洲某部落的第36号王妃",
]

print("=" * 70)
print("🔍 模型预测概率分析")
print("=" * 70)

for text in samples:
    result = validator.validate(text, {})
    print(f"\n文本: {text[:50]}...")
    print(f"预测类别: {result.metadata.get('predicted_class')}")
    print(f"安全概率: {result.metadata.get('safe_probability', 0):.4f}")
    print(f"危险概率: {result.metadata.get('unsafe_probability', 0):.4f}")
    print(f"置信度: {result.metadata.get('confidence', 0):.4f}")
